package com.fxc.crowd.util;

import javax.mail.*;
import javax.mail.internet.InternetAddress;
import javax.mail.internet.MimeMessage;
import java.util.Date;
import java.util.Properties;

/**
 * @author FXC
 * @version 1.0
 * @date 2021/12/5 15:40
 */
/**
 * 使用方法
 *         String Mails[] = {"xxxxxxxxxxxx@qq.com", "xxxxxxxxx@vip.qq.com", "3301633914@qq.com"};//使用的账号，一个QQ可以搞几个账号，主要是腾讯会限制发送频率和数量
 *         String Secret[] = {"zmmhdjgdadaxhjgf"};邮箱账号的密钥（在QQ邮箱账号设置找到并开启SMTP服务）多个依次写，都是一个就写一个
 *         QQMail m=new QQMail("小明",Mails,Secret);
 *         for(int i=0;i<100;i++){
 *             m.sendQQMail("测试","123","3301633914@qq.com");
 *         }
 */
public class QQMail {
    //发件人署名
    private  String NICKNAME;
    //可使用的邮箱账号
    private  String MAILACCOUNTS[] ;
    //账号对应的密钥(多个账号密钥按照账号顺序依次写，全部账号密钥都是同一个就写一个)邮箱账号的密钥（在QQ邮箱账号设置找到并开启SMTP服务）
    private  String MAILSECRET[] ;
    //发送失败后重新尝试发送的最大尝试次数
    private  final int RETRY_LIMIT =3;
    //超时时间（单位毫秒）
    private  final int TIMEOUT =5000;
    private  int MAIL_INDEX = 0;
    private  Properties props;
    private  Session mailSession;
    private  MimeMessage message;
    private  Transport transport;
    //日志（不管你用log4j Logger  注入还是啥做日志，改动下这个方法就行）
    private final void log(String log){
        System.out.println(log);
    }


    /**
     * QQMail
     * @param nickname 邮件署名    字符串
     * @param accounts 所有帐户    字符串数组
     * @param secrets 所以账号对应密钥 字符串数组
     */
    public QQMail(String nickname,String[]accounts,String[]secrets){
        NICKNAME=nickname;
        MAILACCOUNTS=accounts;
        MAILSECRET=secrets;
        init();
    }

    public String getNICKNAME() {
        return NICKNAME;
    }

    public void setNICKNAME(String NICKNAME) {
        this.NICKNAME = NICKNAME;
    }

    public String[] getMAILACCOUNTS() {
        return MAILACCOUNTS;
    }

    public void setMAILACCOUNTS(String[] MAILACCOUNTS) {
        this.MAILACCOUNTS = MAILACCOUNTS;
    }

    public String[] getMAILSECRET() {
        return MAILSECRET;
    }

    public void setMAILSECRET(String[] MAILSECRET) {
        this.MAILSECRET = MAILSECRET;
    }

    public int getRETRY_LIMIT() {
        return RETRY_LIMIT;
    }

    public int getTIMEOUT() {
        return TIMEOUT;
    }

    /**
     * 可以不去主动关闭，反正腾讯那边会主动关闭的
     */
    public void close()  {
        try {
            transport.close();
        } catch (MessagingException e) {
            e.printStackTrace();
        }
    }
    private void init() {
        log("QQ邮件服务初始化开始：账号" + MAILACCOUNTS[MAIL_INDEX]);
        Date start = new Date();
        try {
            // 创建Properties 类用于记录邮箱的一些属性
            props = new Properties();
            // 表示SMTP发送邮件，必须进行身份验证
            props.put("mail.smtp.auth", "true");
            //此处填写SMTP服务器
            props.put("mail.smtp.host", "smtp.qq.com");
            //端口号，QQ邮箱给出了两个端口，但是另一个我一直使用不了，所以就给出这一个587
            props.put("mail.smtp.port", "587");
            // 此处填写你的账号
            props.put("mail.user", MAILACCOUNTS[MAIL_INDEX]);
            // 此处的密码就是前面说的16位STMP口令
            props.put("mail.password", MAILSECRET.length==1?MAILSECRET[0]:MAILSECRET[MAIL_INDEX]);
            //设置超时时间
            props.put("mail.smtp.timeout",""+TIMEOUT);
            // 构建授权信息，用于进行SMTP进行身份验证
            Authenticator authenticator = new Authenticator() {

                protected PasswordAuthentication getPasswordAuthentication() {
                    // 用户名、密码
                    String userName = props.getProperty("mail.user");
                    String password = props.getProperty("mail.password");
                    return new PasswordAuthentication(userName, password);
                }
            };
            // 使用环境属性和授权信息，创建邮件会话
            mailSession = Session.getInstance(props, authenticator);// 创建邮件消息
            message = new MimeMessage(mailSession);
            // 设置发件人
            InternetAddress form = new InternetAddress(
                    props.getProperty("mail.user"), NICKNAME, "utf-8");
            message.setFrom(form);
        } catch (Exception e) {
            e.printStackTrace();
        }
        Date end = new Date();
        log("QQ邮件发送会话初始化成功，耗时" + ((end.getTime() - start.getTime())) + "毫秒");

    }
    /**
     * 切换发件使用的邮箱
     */
    private void changeMailAccount() {
        if (MAIL_INDEX ==MAILACCOUNTS.length-1)
            MAIL_INDEX =0;
        else
            MAIL_INDEX++;
        init();//更换邮箱之后重新初始化
    }

    /**
     * 与QQ邮箱服务器建立连接
     * @throws Exception
     */
    private void connect() throws Exception {
        log("开始建立链接");
        Date start = new Date();
        transport = mailSession.getTransport(new InternetAddress(MAILACCOUNTS[MAIL_INDEX]));
        transport.connect();
        Date end = new Date();
        log("链接已经建立，耗时：" + (end.getTime() - start.getTime()) + "毫秒");
    }
    /**
     * 发送邮件时会阻塞，若使用SpringBoot异步处理
     * 注册线程池代码
     @Configuration
     @EnableAsync  // 启用异步任务
     public class AsyncConfig {

     // 声明一个线程池(并指定线程池的名字)
     @Bean("AsyncThread")
     public Executor asyncExecutor() {
     ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
     //核心线程数5：线程池创建时候初始化的线程数
     executor.setCorePoolSize(3);
     //最大线程数5：线程池最大的线程数，只有在缓冲队列满了之后才会申请超过核心线程数的线程
     executor.setMaxPoolSize(5);
     //缓冲队列500：用来缓冲执行任务的队列
     executor.setQueueCapacity(200);
     //允许线程的空闲时间60秒：当超过了核心线程出之外的线程在空闲时间到达之后会被销毁
     executor.setKeepAliveSeconds(60);
     //线程池名的前缀：设置好了之后可以方便我们定位处理任务所在的线程池
     executor.setThreadNamePrefix("【异步线程】线程");
     executor.initialize();
     return executor;
     }
     }
     */
    /**
     * 发送邮件方法
     * @param title 邮件标题
     * @param html_content 邮件内容（支持html，图片等内容可能会被拦截，需要用户点击查看才能看到，或者让用户设置信任这个邮箱）
     * @param receiver 收件人邮箱
     */
    //@Async("AsyncThread")//异步，使用线程池的时候用这个标明这个方法由线程池处理
    public synchronized void sendQQMail(String title, String html_content, String receiver) {
        try {
            if (transport == null || !transport.isConnected())
                connect();
            // 设置收件人的邮箱
            message.setRecipient(Message.RecipientType.TO, new InternetAddress(receiver));
            // 设置邮件标题
            message.setSubject(title, "utf-8");
            // 设置邮件的内容体
            message.setContent(html_content, "text/html;charset=UTF-8");
            Date start = new Date();
            try {
                //保存修改
                message.saveChanges();
                //发送邮件
                transport.sendMessage(message, new InternetAddress[]{new InternetAddress(receiver)});
                log("使用邮箱："+ MAILACCOUNTS[MAIL_INDEX] +"发送邮件给"+receiver+"\n标题：\n"+title+"\n内容：\n"+html_content);
            } catch (Exception e) {
                //由于被腾讯方面因超时被关闭连接属于正常情况
                log("邮件发送失败，正在尝试和QQ邮件服务器重新建立链接");
                boolean success=false;
                for (int i = 1; i<= RETRY_LIMIT; i++)
                    try {
                        connect();
                        log("使用邮箱："+ MAILACCOUNTS[MAIL_INDEX] +"成功建立链接");
                        success=true;
                        break;
                    } catch (Exception ee) {
                        changeMailAccount();
                        log("链接建立失败,切换到邮箱："+ MAILACCOUNTS[MAIL_INDEX] +"，进行第"+i+"次重试...");
                    }
                if (success)
                {
                    message.saveChanges();
                    transport.sendMessage(message, new InternetAddress[]{new InternetAddress(receiver)});
                    log("重建链接后使用邮箱："+ MAILACCOUNTS[MAIL_INDEX] +"发送邮件给"+receiver+"\n标题：\n"+title+"\n内容：\n"+html_content);
                }
                else
                {
                    log("链接多次尝试后无法建立，邮件发送失败！备注：收信人"+receiver);
                    return;
                }
            }
            Date end = new Date();
            log("成功发送邮件给" + receiver +"标题："+title+ ",耗时" + ((end.getTime() - start.getTime())) + "毫秒");
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}


